import glob
import os
import time
import shutil
import pandas as pd
from 筛选excel import excel
from DrissionPage.common import Keys

from 获取无效id import get_wuxiaoid


def get_data_only(page, down_path, date):
    shop_name = '抖音小店-' + page(".index_userName__16Isl").text
    if os.path.exists(f"C:\\抖音客服数据\\{date}\\{shop_name}.xlsx"):
        return
    page('接待').click()
    time.sleep(8)
    # 切换到最新打开的标签页
    tab = page.get_tab(page.latest_tab)
    if tab.s_eles("退出引导"):
        tab("退出引导").click()
        time.sleep(1)
    if tab.s_eles('.qg8dJkHI2UbLch1KK8J6'):
        tab('.qg8dJkHI2UbLch1KK8J6').ele("tag:img").click()
        time.sleep(1)
    tab("数据").click()
    tab('历史会话').click()
    tab.wait(2)
    tab.eles('.Mvg3OH8ZtQ7l4K_FCoZv')[4].click()
    tab('.sp-select-all').click()
    ele = tab('.rc-virtual-list-holder-inner')
    ele.ele('人工会话，用户主动离开').click()
    ele.ele('人工会话，客服结束会话').click()
    tab.eles('.Mvg3OH8ZtQ7l4K_FCoZv')[4].click()
    tab.wait(1)
    # 取消勾选未评价
    tab.eles('.Mvg3OH8ZtQ7l4K_FCoZv')[5].ele('.auxo-select-selection-item-remove').click()
    # 会话时间‘工作时间’
    tab.eles('.Mvg3OH8ZtQ7l4K_FCoZv')[7].click()
    tab.wait(1)
    tab.eles('.rc-virtual-list-holder-inner')[1].ele('工作时间').click()
    # 选择人工接待‘是’
    tab.eles('.Mvg3OH8ZtQ7l4K_FCoZv')[9].click()
    tab.wait(1)
    tab.eles('.rc-virtual-list-holder-inner')[2].ele('是').click()
    # 输入时间
    tab('@placeholder=开始日期').input(date)
    tab('@placeholder=结束日期').input(date)
    tab.actions.key_down('ENTER')
    tab('查询').click()
    tab.wait(1)
    if tab.s_ele('暂无数据'):
        print('暂无数据')
        df = pd.DataFrame()
        df.to_excel(f"C:\\抖音客服效率统计\\{date}\\{shop_name}.xlsx")
        return
    tab('导出报表').click(by_js=True)
    tab.wait(3)
    tab('@placeholder=请输入文件名').input(date)
    tab('确定').click()
    tab.wait.download_begin()
    tab.wait.downloads_done()  # 等待下载结束
    wuxiaoid_list = get_wuxiaoid(tab)
    # 获取文件夹中所有的.xlsx文件
    xlsx_files = glob.glob(os.path.join(down_path, "*.xlsx"))
    os.makedirs(f'C:\\抖音客服效率统计\\{date}', exist_ok=True)
    shutil.move(xlsx_files[0], f"C:\\抖音客服效率统计\\{date}\\{shop_name}.xlsx")
    excel(shop_name, date, f"C:\\抖音客服效率统计\\{date}\\{shop_name}.xlsx", wuxiaoid_list)
